This banner text can have markup.
English | 中文
MegEngine is a fast, scalable and easy-to-use numerical evaluation framework, with auto-differentiation.
NOTE: MegEngine now only supports Linux platform with Python 3.5 or higher. On Windows 10 you could try WSL(Windows Subsystem for Linux) to use Linux within Windows.
Commands to install from binaries via pip wheels are as follows:
bashpip3 install megengine -f https://megengine.org.cn/whl/mge.html
Most of the dependencies of MegEngine are located in
third_party directory, and you donot need to install these by yourself. you can prepare these repositories by executing:
But some dependencies should be manually installed:
Out-Of-Source flavor, and compile in a
mostly-static way.Here are the instructions:
Make a directory for the build.
bashmkdir -p buildcd build
Generate build configurations by
For CUDA build:
bashcmake .. -DMGE_WITH_TEST=ON
For CPU only build, use
bashcmake .. -DMGE_WITH_CUDA=OFF -DMGE_WITH_TEST=ON
For deployment with C++ only, use
-DMGE_INFERENCE_ONLY=ON, and turn off test with
bashcmake .. -DMGE_INFERENCE_ONLY=ON -DMGE_WITH_TEST=OFF
-DCMAKE_INSTALL_PREFIX=YOUR_PATH to specify the install path.
Start to build.
[optional] Install the library if compiled for deployment at step 2.
Here are some other useful options for the build.
MGE_ARCHspecifies which arch MegEngine are building for. (default AUTO)
MGE_WITH_DISTRIBUTEDif multiple machine distributed support is enabled. (default ON)
MGE_WITH_PYTHON_MODULEif build python module. (default ON)
OpenBLASas BLAS library for MegEngine. (default
nvcc. (default not supply)
MGE_DISABLE_FLOAT16if disable float16 support. (default OFF)
MGE_ENABLE_EXCEPTIONSif enable exception support in C++. (default ON)
MGE_ENABLE_LOGGINGif enable logging in MegEngine. (default AUTO)
More options can be found by:
bashcd buildcmake -LAH .. 2>/dev/null| grep -B 1 'MGE_' | less
We believe we can build an open and friendly community and power humanity with AI.
MegEngine is Licensed under the Apache License, Version 2.0
Copyright (c) 2014-2020 Megvii Inc. All rights reserved.
git clone MegEngine-MegEngine_-_2020-03-26_08-46-30.bundle